package com.leetcode_mianshi_150.figure;

import com.sun.org.apache.bcel.internal.generic.LCONST;
import org.junit.Test;

import java.util.*;

/**
 * @author: 临晖
 * @since: 2024-04-27
 */
public class Solution {
    /**
     * 133. 克隆图
     * 中等
     * 相关标签
     * 相关企业
     * 给你无向 连通 图中一个节点的引用，请你返回该图的 深拷贝（克隆）。
     * 图中的每个节点都包含它的值 val（int） 和其邻居的列表（list[Node]）。
     * class Node {
     * public int val;
     * public List<Node> neighbors;
     * }
     * 测试用例格式：
     * <p>
     * 简单起见，每个节点的值都和它的索引相同。例如，第一个节点值为 1（val = 1），第二个节点值为 2（val = 2），以此类推。该图在测试用例中使用邻接列表表示。
     * <p>
     * 邻接列表 是用于表示有限图的无序列表的集合。每个列表都描述了图中节点的邻居集。
     * <p>
     * 给定节点将始终是图中的第一个节点（值为 1）。你必须将 给定节点的拷贝 作为对克隆图的引用返回。
     *
     * @param node
     * @return
     */
    public Node cloneGraph(Node node) {
        //方法一：dfs
        //Map<Node, Node> lookUp = new HashMap<>();
        //return dfs(node, lookUp);

        //方法二：bfs
        if (node == null) {
            return node;
        }
        LinkedList<Node> que = new LinkedList<>();
        Map<Node, Node> lookUp = new HashMap<>();

        Node clone = new Node(node.val, new ArrayList<>());
        lookUp.put(node, clone);
        que.add(node);

        while (!que.isEmpty()) {
            Node temp = que.removeFirst();
            for (Node no : temp.neighbors) {
                if (!lookUp.containsKey(no)) {
                    clone = new Node(no.val, new ArrayList<>());
                    lookUp.put(no, clone);
                    que.addLast(no);
                }

                lookUp.get(temp).neighbors.add(lookUp.get(no));
            }
        }

        return lookUp.get(node);
    }

    private Node dfs(Node node, Map<Node, Node> lookUp) {
        if (node == null) {
            return node;
        }

        if (lookUp.containsKey(node)) {
            return lookUp.get(node);
        }

        Node clone = new Node(node.val, new ArrayList<>());
        lookUp.put(node, clone);
        for (Node no : node.neighbors) {
            clone.neighbors.add(dfs(no, lookUp));
        }

        return clone;
    }






    @Test
    public void main() {
        Node node1 = new Node(1);
        Node node2 = new Node(2);
        Node node3 = new Node(3);
        Node node4 = new Node(4);

        node1.neighbors = Arrays.asList(node2, node4);
        node2.neighbors = Arrays.asList(node1, node3);
        node3.neighbors = Arrays.asList(node2, node4);
        node4.neighbors = Arrays.asList(node1, node3);

        cloneGraph(node1);
    }


}
